import requests


# 定义 类
class IhrmEmpApi(object):
    # 添加 实例属性
    def __init__(self):
        # 添加属性 url
        self.url = "http://ihrm-test.itheima.net/api/sys/user/"

    # 定义方法 - 添加员工
    def add_emp(self, req_header, req_body):
        resp = requests.post(url=self.url, headers=req_header, json=req_body)
        return resp

    # 查询员工
    def query_emp(self, emp_id, req_header):
        resp = requests.get(url=self.url + emp_id, headers=req_header)
        return resp

    # 修改员工
    def modify_emp(self, emp_id, req_header, req_body):
        resp = requests.put(url=self.url + emp_id, headers=req_header, json=req_body)
        return resp

    # 删除员工
    def del_emp(self, emp_id, req_header):
        return requests.delete(url=self.url + emp_id, headers=req_header)


if __name__ == '__main__':
    # 创建 IhrmEmpApi 实例（对象）
    ihrm_emp_api = IhrmEmpApi()

    # 准备 请求头
    req_header = {"Authorization": "2950d3a2-e7e8-4654-9fe6-ee3aa3eb44c0"}

    # 准备 请求体
    req_body = {
        "username": "user87655430",
        "mobile": "17257483970",
        "workNumber": "9527890aa"
    }
    # # 使用 实例，调用方法 - 添加
    # resp = ihrm_emp_api.add_emp(req_header, req_body)
    # print("添加员工：", resp.json())

    # 使用 实例，调用方法 - 查询
    emp_id = "1522203507589971968"
    resp = ihrm_emp_api.query_emp(emp_id, req_header)
    print("查询员工：", resp.json())

    modify_req_body = {"username": "至尊宝9527"}
    # 使用 实例，调用方法 - 修改
    resp = ihrm_emp_api.modify_emp(emp_id, req_header, modify_req_body)
    print("修改员工：", resp.json())

    # 使用 实例，调用方法 - 删除
    resp = ihrm_emp_api.del_emp(emp_id, req_header)
    print("删除员工：", resp.json())